import matplotlib.pyplot as plt
import numpy as np
from scipy import signal

n_min = 0
n_max = 20
n = np.arange(n_min, n_max + 1, 1)
n_y = np.arange(n_min * 2, n_max * 2 + 1, 1)

x = 0.5 ** n
h = 0.8 ** n

y = signal.convolve(x, h)

plt.subplot(311)
plt.stem(n, x)
plt.xlabel('n')
plt.ylabel('x[n]')
plt.subplot(312)
plt.stem(n, h)
plt.xlabel('n')
plt.ylabel('h[n]')
plt.subplot(313)
plt.stem(n_y, y)
plt.xlabel('n')
plt.ylabel('y[n]')
plt.show()
